from langchain.chains.llm import LLMChain
from langchain_core.prompts import PromptTemplate, HumanMessagePromptTemplate, ChatPromptTemplate

from ChatGLM_new import xinghuo_llm

print("################################chain##############################")
prompt = PromptTemplate(
    input_variables=["product"],
    template="""
    你是一个销售专家，请根据产品名称，给出产品简介。
    产品名称: {product}
    产品简介:
    """,
)
chain = LLMChain(prompt=prompt, llm=xinghuo_llm)
#res = chain.invoke({"product": "智能机器人"})
#print(res)


print("################################多个提示词##############################")
input_list = [
    {"product": "袜子"},
    {"product": "电脑"},
    {"product": "男鞋"}
]
#res = chain.apply(input_list)
#print(res)



print("################################聊天模型链##############################")
human_message_prompt = HumanMessagePromptTemplate(
    prompt=PromptTemplate(
        template="我希望你能充当新公司的命名顾问。一个生产{product}的公司好的中文名字是什么,请给出五个选项？?",
        input_variables=["product"],))
chat_prompt_template = ChatPromptTemplate.from_messages([human_message_prompt])
chain_chat = LLMChain(llm=xinghuo_llm, prompt=chat_prompt_template)
res = chain_chat.invoke({"product": "彩色袜子"})
print(res)


